package com.code.test.first.hash;

import java.util.HashSet;
import java.util.Set;

/**
 * https://github.com/youngyangyang04/leetcode-master/blob/master/problems/0202.%E5%BF%AB%E4%B9%90%E6%95%B0.md
 * 编写一个算法来判断一个数 n 是不是快乐数。
 * <p>
 * 「快乐数」定义为：对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和，然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。如果 可以变为  1，那么这个数就是快乐数。
 * <p>
 * 如果 n 是快乐数就返回 True ；不是，则返回 False 。
 */
public class Code202 {

    public static void main(String[] args) {
        boolean s = isHappy(7);
        System.out.println(s);
    }

    public static boolean isHappy(int n) {
        //记录已经遍历过的数
        Set<Integer> record = new HashSet<>();
        while (n != 1 && !record.contains(n)) {
            record.add(n);
            n = getNext(n);
        }
        return n == 1;
    }

    public static int getNext(int n) {
        int ret = 0;
        while (n > 0) {
            int temp = n % 10;
            ret += temp * temp;
            n = n / 10;
        }
        return ret;
    }

}
